# 打开文件并读取内容
with open("aa.txt", "r", encoding="utf-8") as f:
    lines = f.readlines()

# 创建一个空列表，用于存储补全后的 SQL 语句
updated_lines = []

# 遍历文件中的每一行
for line in lines:
    # 检查是否为 UPDATE 语句
    if line.startswith("UPDATE SYS_ORGANIZATION SET"):
        # 查找 SET 子句后面的 ORG_CODE 值
        set_index = line.find("ORG_CODE='") + len("ORG_CODE='")
        end_index = line.find("'", set_index)
        org_code = line[set_index:end_index]

        # 查找 WHERE 子句
        where_index = line.find("WHERE")

        # 如果 WHERE 子句后面没有内容，则补全 ORG_CODE 值
        if where_index != -1 and line[where_index+6:].strip() == ";":
            # 将补全后的 SQL 语句添加到列表中
            updated_lines.append(line[:where_index+6] + f"ORG_CODE='{org_code}';\n")
        else:
            # 将原始 SQL 语句添加到列表中
            updated_lines.append(line)
    else:
        # 将非 UPDATE 语句添加到列表中
        updated_lines.append(line)

# 将补全后的 SQL 语句写入新的文件
with open("updated_aa.txt", "w", encoding="utf-8") as f:
    f.writelines(updated_lines)
